import matplotlib.pyplot as plt
import numpy as np
from scipy import signal


n = np.array([-2, -1, 0, 1, 2, 3, 4, 5])
x = np.array([0, 1, 2, 1, 1, 0, 0, 0])
h = np.array([0, 0, 0, -1, 0, 0, -1, 1])
n_y = np.array([-4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

y = signal.convolve(x, h)

plt.subplot(311)
plt.stem(n, x)
plt.xlabel('n')
plt.ylabel('x[n]')

plt.subplot(312)
plt.stem(n, h)
plt.xlabel('n')
plt.ylabel('h[n]')

plt.subplot(313)
plt.stem(n_y, y)
plt.xlabel('n')
plt.ylabel('y[n]')

plt.show()
